package com.neusoft;

/**
 * @Author lichunlong lcl253137@alibaba-inc.com
 * @Desc 稀疏数组代码示例: 11行x11列的棋盘，颜色分别为黑子和蓝子
 * @Date 2024-03-27
 */
public class SparseArray {

    public static void main(String[] args) {
        // 1、初始化生成棋盘的二维数组
        int[][] chessArr = initChessArr();

        // 2、输出原始的二维数组
        System.out.println("输出原始的二维数组");
        printChessArr(chessArr);

        // 3、二维数组 => 稀疏数组
        // 3-1、先获取稀疏数组中非0数据的个数
        int chessArrValidNum = calChessArrValidNum(chessArr);
        // 3-2、创建对应的稀疏数组并赋值
        int[][] sparseArr = initSparseArr(chessArr, chessArrValidNum);
        // 3-3、打印稀疏数组
        System.out.println("输出转换后的稀疏数组");
        printSparseArr(sparseArr);

        // 4、稀疏数组 => 二维数组
        int[][] originArr = sparse2OriginArr(sparseArr);
        System.out.println("输出还原后的二维数组");
        printChessArr(originArr);
    }

    /**
     * 初始化生成棋盘的二维数组
     *
     * @return 棋盘的二维数组
     */
    public static int[][] initChessArr() {
        /*
         * 创建一个原始的二维数组 11 * 11
         * 0: 表示没有棋子， 1: 表示黑子 2: 表示蓝子
         */
        int[][] chessArr = new int[11][11];
        chessArr[1][2] = 1;
        chessArr[2][3] = 2;
        chessArr[4][5] = 2;

        return chessArr;
    }

    /**
     * 初始化稀疏数组
     *
     * @param chessArr         原数组
     * @param chessArrValidNum 原数组非0有效数据的个数
     * @return 稀疏数组
     */
    public static int[][] initSparseArr(int[][] chessArr, int chessArrValidNum) {
        int[][] sparseArr = new int[chessArrValidNum + 1][3];
        // 1、给稀疏数组的第一列赋默认值: 原数组X轴、原数组Y抽、原数组非0有效数据的个数
        sparseArr[0][0] = 11;
        sparseArr[0][1] = 11;
        sparseArr[0][2] = chessArrValidNum;

        /*
         * 2、给稀疏数组赋值
         * count: 用于记录是第几个非0数据
         */
        int count = 0;
        for (int i = 0; i < chessArr.length; i++) {
            for (int j = 0; j < chessArr[i].length; j++) {
                if (chessArr[i][j] != 0) {
                    count++;
                    sparseArr[count][0] = i;
                    sparseArr[count][1] = j;
                    sparseArr[count][2] = chessArr[i][j];
                }
            }
        }

        return sparseArr;
    }

    /**
     * 获取稀疏数组中非0数据的个数
     *
     * @param chessArr 原始棋盘的二维数组
     */
    public static int calChessArrValidNum(int[][] chessArr) {
        int sum = 0;
        for (int[] row : chessArr) {
            for (int data : row) {
                if (data != 0) {
                    sum++;
                }
            }
        }

        return sum;
    }

    /**
     * 打印原始的棋盘数据
     *
     * @param chessArr 原始的二维数组
     */
    public static void printChessArr(int[][] chessArr) {
        for (int[] row : chessArr) {
            for (int data : row) {
                System.out.printf("%d\t", data);
            }
            System.out.println();
        }
    }

    /**
     * 打印转换后的稀疏数据
     *
     * @param sparseArr 转换后的稀疏数据
     */
    public static void printSparseArr(int[][] sparseArr) {
        for (int[] row : sparseArr) {
            System.out.printf("%d\t%d\t%d\t\n", row[0], row[1], row[2]);
        }
        System.out.println();
    }

    /**
     * 稀疏数组 => 二维数组
     *
     * @param sparseArr 稀疏数组
     * @return 二维数组
     */
    public static int[][] sparse2OriginArr(int[][] sparseArr) {
        // 1、先读取稀疏数组的第一行，根据第一行的数据，创建原始的二维数组
        int[][] originArr = new int[sparseArr[0][0]][sparseArr[0][1]];

        // 2、在读取稀疏数组后几行的数据(从第二行开始)，并赋给原始的二维数组即可
        for (int i = 1; i < sparseArr.length; i++) {
            originArr[sparseArr[i][0]][sparseArr[i][1]] = sparseArr[i][2];
        }

        return originArr;
    }
}
